Fixing stupid typo; Everything working as expected for device code