Add exception for tests