Implemeted should module in tests