Fixes issue with controller

Fixed an issue where pkeys was not updating properly inside Input