Updated tests for customer