Fixing issues with knockout