Small style clean up