Make it work with h2 database