Fix tests for real device