adapted test to completed implementation