Late Shift: Minor Update

Minor bug fixes