Fixed the dependency problem (test vs real)