Fixed the problem with wrong color