Don't need this in here
Just reuse the class for testing