Adding snippet to get tests passing