Small Input fix