Updated revision and readded custom test