Updated component unit tests