Fix for custom input not working