Changed tests to reflect API