Fix Component unit tests