diff --git a/models/book_test.go b/models/book_test.go
index 2053a9d..b9ff7c7 100644
--- a/models/book_test.go
+++ b/models/book_test.go
@@ -97,11 +97,11 @@ func TestAddOrUpdateBook(t *testing.T) {
 	assert.Equal(t, testbook.Title, book1updated.Title)
 
 	// Delete the book
-	err = DeleteBookByID(1)
+	err = DeleteBookByID(book1.ID)
 	assert.NoError(t, err)
 
 	// Check if its gone
-	_, exists, err = GetBookByID(1)
+	_, exists, err = GetBookByID(book1.ID)
 	assert.NoError(t, err)
 	assert.False(t, exists)
 }