removes unnecessary codes
fixes an argument type
tiny fix