Fix type annotation for floor'

0fc9ec28ef446c135c3c00dfb646350af00b21eb broke the types, this fixes it.