remove unneeded return and fix os import naming