Moved tests to internal package